package leetcode.hash;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 第18题. 四数之和
 * 给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。
 * 请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）
 * 示例 1：
 * <p>
 * 输入：nums = [1,0,-1,0,-2,2], target = 0
 * 输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
 * 示例 2：
 * <p>
 * 输入：nums = [2,2,2,2,2], target = 8
 * 输出：[[2,2,2,2]]
 */
public class _18 {

    public static void main(String[] args) {
        int[] nums = new int[]{1000000000, 1000000000, 1000000000, 1000000000};
        List<List<Integer>> lists = fourSum(nums, -294967296);
        System.out.println("lists = " + lists);

    }

    public static List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);

        List<List<Integer>> result = new ArrayList<>();

        for (int aIndex = 0; aIndex < nums.length - 3; aIndex++) {
//          target 有可能小于0 这段判断不能要
//            if (nums[aIndex] > target) {
//                return result;
//            }
            // 给a去重
            if (aIndex > 0 && nums[aIndex] == nums[aIndex - 1]) {
                continue;
            }

            for (int bIndex = aIndex + 1; bIndex < nums.length - 2; bIndex++) {
//               target 有可能小于0 这段判断不能要
//                if (nums[aIndex] + nums[bIndex] > target) {
//                    break;
//                }
                // 给b去重
                if (bIndex > aIndex + 1 && nums[bIndex] == nums[bIndex - 1]) {
                    continue;
                }

                int cIndex = bIndex + 1;
                int dIndex = nums.length - 1;
                while (cIndex < dIndex) {
                    // c去重
                    if (cIndex > bIndex + 1 && nums[cIndex] == nums[cIndex - 1]) {
                        cIndex++;
                        continue;
                    }
                    // d去重
                    if (dIndex < nums.length - 1 && nums[dIndex] == nums[dIndex + 1]) {
                        dIndex--;
                        continue;
                    }
                    long sum = (long) nums[aIndex] + (long) nums[bIndex] + (long) nums[cIndex] + (long) nums[dIndex];
                    if (sum > target) {
                        dIndex--;
                    } else if (sum < target) {
                        cIndex++;
                    } else {
                        List<Integer> res = new ArrayList<>();
                        res.add(nums[aIndex]);
                        res.add(nums[bIndex]);
                        res.add(nums[cIndex]);
                        res.add(nums[dIndex]);
                        result.add(res);
                        // 默认 c往前移动
                        cIndex++;
                    }
                }

            }

        }
        return result;
    }
}
